package reference;

import com.hankcs.hanlp.seg.common.Term;
import com.hankcs.hanlp.tokenizer.NLPTokenizer;

import java.util.ArrayList;
import java.util.List;

/**
 * @author hujian
 * @date 2023/2/18 8:44
 * @description
 */
public class HanLPUtil {

    public static List<String> getNames(String sentence) {

//        sentence = RefUtils.getCnTxtWithSpace(sentence);

        System.out.println(sentence);

        List<Term> termList = NLPTokenizer.segment(sentence);

        System.out.println(termList);

        List<String> names = new ArrayList<>();
        for (Term term : termList) {
            if ("nr".equals(term.nature.toString())) {
                names.add(term.word);
            }
        }
        return names;
    }


    public static void main(String[] args) {
        String text = "徐峥嵘 浅议高校基建工程管理存在的问题及对策[J]";
        System.out.println(getNames(text));
        text = "16.王力淑.高校基本建设管理问题研究——以S大学为个案分析[D]";
        System.out.println(getNames(text));
        text = "15.尤元宝.协同管理模式下高校基建项目管理研究[J]";
        System.out.println(getNames(text));
        text = "14.陈国柱.中国美术学院基建项目管理模式优选研究[D]";
        System.out.println(getNames(text));
    }
}
